{- This file is part of funbot.
 -
 - Written in 2016, 2017 by fr33domlover <fr33domlover@riseup.net>.
 -
 - ♡ Copying is an act of love. Please copy, reuse and share.
 -
 - The author(s) have dedicated all copyright and related and neighboring
 - rights to this software to the public domain worldwide. This software is
 - distributed without any warranty.
 -
 - You should have received a copy of the CC0 Public Domain Dedication along
 - with this software. If not, see
 - <http://creativecommons.org/publicdomain/zero/1.0/>.
 -}

module FunBot.Locations
    ( lookupLocal
    , lookupGlobal
    , lookupBoth
    )
where

import Database.Persist
import Network.IRC.Fun.Types (Channel)

import FunBot.Model
import FunBot.Types
import FunBot.Util

lookupLocal :: Channel -> LocationLabel -> BotSession (Maybe LocationText)
lookupLocal chan label = runDB $ do
    mloc <- getBy $ UniqueLocalLocation chan label
    return $ localLocationContent . entityVal <$> mloc

lookupGlobal :: LocationLabel -> BotSession (Maybe LocationText)
lookupGlobal label = do
    mloc <- runDB $ getBy $ UniqueGlobalLocation label
    return $ globalLocationContent . entityVal <$> mloc

lookupBoth :: Channel -> LocationLabel -> BotSession (Maybe LocationText)
lookupBoth chan label = do
    mloc <- lookupLocal chan label
    maybe (lookupGlobal label) (return . Just) mloc
